#pragma once
#include <iostream>
#include <vector>
#include <pthread.h>
#include <memory>
#include <functional>
#include <queue>
#include "Log.hpp"
#include "Mutex.hpp"
#include "Cond.hpp"
#include "Thread.hpp"

namespace ThreadPoolModule
{
    using namespace LockModule;
    using namespace CondModule;
    using namespace LogModule;
    using namespace ThreadModule;

    using thread_t = std::shared_ptr<Thread>;
    const static int defaultnum = 5;

    template<class T>
    class ThreadPool
    {
    private:
        bool IsEmpty() { return _taskq.empty(); }

        void HandlerTask(std::string name)
        {
            LOG(LogLevel::INFO) << "线程: " << name << ", 进入HandlerTask的逻辑";
            while (true)
            {
                // 拿任务
                T t;
                {
                    LockGuard lockguard(_mtx);
                    while (IsEmpty() && _Isrunning)
                    {
                        _wait_num++;
                        _cond.Wait(_mtx);
                        _wait_num--;
                        LOG(LogLevel::INFO) << "wait num : " << _wait_num;
                    }
                    // 任务队列为空 && 线程池退出了
                    if (IsEmpty() && !_Isrunning)
                        break;

                    t = _taskq.front();
                    _taskq.pop();

                }

                // 处理任务
                t(name);
            }
            LOG(LogLevel::INFO) << "线程: " << name << " 退出";
        }

    public:
        ThreadPool(int num = defaultnum)
            : _num(num),
              _wait_num(0),
              _Isrunning(false)
        {
            for (int i = 0; i < _num; i++)
            {
                //_threads.push_back(std::make_shared<Thread>(std::bind(&ThreadPool::HandlerTask, this, std::placeholders::_1)));
                _threads.push_back(std::make_shared<Thread>([this](std::string name){
                    HandlerTask(name);
                }));
                LOG(LogLevel::INFO) << "构建线程" << _threads.back()->Name() << "对象 ... 成功";
            }
        }
        template <class K>
        void Enqueue(K&& in)
        {
            LockGuard lck(_mtx);
            if (!_Isrunning)
                return;
            _taskq.push(std::forward<K>(in));
            if (_wait_num)
            {
                _cond.Notify();
            }
        }
        // 不能用类模版参数来写,因为类被实例化的时候就T就已经确定了，不能被推导，就不能触发引用折叠
        // 引用折叠（Reference Collapsing）仅在以下场景发生：
        // 模板参数推导 的上下文中（例如函数模板的 T&& 形参）。
        // 类型别名或 decltype 中显式组合引用时（如 typedef T& &&）。
        // 若 T 未经过推导（例如直接指定类型），则 T&& 是普通的右值引用，不触发引用折叠。
        // void Enqueue(T&& in)
        // {
        //     LockGuard lck(_mtx);
        //     if (!_Isrunning)
        //         return;
        //     _taskq.push(std::forward<T>(in));
        //     if (_wait_num)
        //     {
        //         _cond.Notify();
        //     }
        // }
        void Start()
        {
            if (_Isrunning)
                return;
            _Isrunning = true; 
            for (auto &thread_ptr : _threads)
            {
                thread_ptr->Name();
                LOG(LogLevel::INFO) << "启动线程" << thread_ptr->Name() << " ... 成功";
                thread_ptr->Start(); 
            }
        }
        void Wait()
        {
            for (auto &thread_ptr : _threads)
            {
                thread_ptr->Join();
                LOG(LogLevel::INFO) << "回收线程" << thread_ptr->Name() << " ... 成功";
            }
        }
        void Stop()
        {
            LockGuard lockguard(_mtx);
            if(_Isrunning)
            {
                _Isrunning = false; 
                // 线程自己退出(要唤醒) && 历史的任务被处理完了
                if(_wait_num > 0)
                    _cond.NotifyAll();
            }
        }
        ~ThreadPool()
        {
        }

    private:
        std::vector<thread_t> _threads;
        int _num;
        int _wait_num;
        std::queue<T> _taskq;

        Mutex _mtx;
        Cond _cond;
        bool _Isrunning;
    };
}